One step list operation in Python